package com.topone.Utils;

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.geom.AffineTransform;
import java.awt.image.AffineTransformOp;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

import javax.imageio.ImageIO;

import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.WriterException;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;

/**
 * 二维码生成和读的工具类
 */
public class CreateQReCodeUtils {

	// 二维码中间的LOGO图片设置
	private static final int IMAGE_WIDTH = 90; // 宽度
	private static final int IMAGE_HEIGHT = 90; // 高度
	private static final int IMAGE_HALF_WIDTH = IMAGE_WIDTH / 2;
	private static final int FRAME_WIDTH = 3; // 距离二维码
	private static final int IMAGE_PADDING=5;//LOGO边框距图片距离
 	private static final int IMAGE_BORDER=5;//LOGO边框宽度
 	private static final String CHARACTER="utf-8";
	// 二维码写码器
	private static MultiFormatWriter mutiWriter = new MultiFormatWriter();
 
	/**
	 * 添加在指定位置追加图片水印
	 * 
	 * @param waterImage
	 *            --水印文件地址
	 * @param img
	 *            -- 目标
	 * @param x
	 *            --x坐标
	 * @param y
	 *            --y坐标
	 */
	public final static BufferedImage addWaterImage(BufferedImage waterImage, BufferedImage img, int x, int y) {
		try {
			// 目标文件
			int width = img.getWidth(null);
			int height = img.getHeight(null);
			//File _filebiao = new File(waterImage);
			//BufferedImage image = ImageIO.read(waterImage);
			Graphics g = waterImage.createGraphics();
			g.drawImage(img, x, y, width, height, null);
			g.dispose();
			return waterImage;
		} catch (Exception e) {
			e.printStackTrace();
			return null;
		}
	}
 
	/**
	 * 添加在指定位置追加文字水印
	 * 
	 * @param text
	 *            --要添加的文字
	 * @param img
	 *            --目标
	 * @param fontName
	 *            --字体
	 * @param fontStyle
	 *            --样式
	 * @param color
	 *            --字体颜色
	 * @param fontSize
	 *            --字体大小
	 * @param x
	 *            --x坐标
	 * @param y
	 *            --y坐标
	 * @return
	 */
	public static BufferedImage addWaterText(String text, BufferedImage img, String fontName, int fontStyle, Color color, int fontSize, int x, int y) {
		try {
 
			// 目标文件
			int width = img.getWidth(null);
			int height = img.getHeight(null);
			Graphics g = img.createGraphics();
			g.drawImage(img, 0, 0, width, height, null);
			g.setColor(color);
			g.setFont(new Font(fontName, fontStyle, fontSize));
			g.drawString(text, x, y + fontSize);
			g.dispose();
			return img;
		} catch (Exception e) {
			System.out.println(e);
			return null;
		}
	}
 
	/**
	 * 
	 * 生成对应的二维码
	 * 
	 * @param text
	 *            --二维码内容
	 * @param width
	 *            --二维码图片宽度
	 * @param height
	 *            --二维码图片高度
	 * @param srcImagePath
	 *            -- logo图片地址
	 * @return
	 * @throws WriterException
	 * @throws IOException
	 */
	public static BufferedImage createQR(String text, int width, int height, BufferedImage srcImage) throws WriterException, IOException {
		// 读取源图像
		BufferedImage scaleImage = scale(srcImage, IMAGE_WIDTH, IMAGE_HEIGHT, true,IMAGE_BORDER,IMAGE_PADDING);
		
		int[][] srcPixels = new int[IMAGE_WIDTH][IMAGE_HEIGHT];
		for (int i = 0; i < scaleImage.getWidth(); i++) {
			for (int j = 0; j < scaleImage.getHeight(); j++) {
				srcPixels[i][j] = scaleImage.getRGB(i, j);
			}
		}
 
		Map<EncodeHintType, Object> hint = new HashMap<EncodeHintType, Object>();
		hint.put(EncodeHintType.CHARACTER_SET, CHARACTER);
		hint.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.H);
		hint.put(EncodeHintType.MARGIN, 0);
		// 生成二维码
		BitMatrix matrix = mutiWriter.encode(text, BarcodeFormat.QR_CODE, width, height, hint);
 
		// 二维矩阵转为一维像素数组
		int halfW = matrix.getWidth() / 2;
		int halfH = matrix.getHeight() / 2;
		int[] pixels = new int[width * height];
 
		for (int y = 0; y < matrix.getHeight(); y++) {
			for (int x = 0; x < matrix.getWidth(); x++) {
				// 读取图片
				if (x > halfW - IMAGE_HALF_WIDTH && x < halfW + IMAGE_HALF_WIDTH && y > halfH - IMAGE_HALF_WIDTH && y < halfH + IMAGE_HALF_WIDTH) {
					pixels[y * width + x] = srcPixels[x - halfW + IMAGE_HALF_WIDTH][y - halfH + IMAGE_HALF_WIDTH];
				}
				// 在图片四周形成边框
				else if ((x > halfW - IMAGE_HALF_WIDTH - FRAME_WIDTH && x < halfW - IMAGE_HALF_WIDTH + FRAME_WIDTH && 
						y > halfH - IMAGE_HALF_WIDTH - FRAME_WIDTH && y < halfH + IMAGE_HALF_WIDTH + FRAME_WIDTH) ||
						(x > halfW + IMAGE_HALF_WIDTH - FRAME_WIDTH && x < halfW + IMAGE_HALF_WIDTH + FRAME_WIDTH && 
								y > halfH - IMAGE_HALF_WIDTH - FRAME_WIDTH && y < halfH + IMAGE_HALF_WIDTH + FRAME_WIDTH) || 
						(x > halfW - IMAGE_HALF_WIDTH - FRAME_WIDTH && x < halfW + IMAGE_HALF_WIDTH + FRAME_WIDTH && y > halfH - IMAGE_HALF_WIDTH - FRAME_WIDTH &&
								y < halfH - IMAGE_HALF_WIDTH + FRAME_WIDTH) || (x > halfW - IMAGE_HALF_WIDTH - FRAME_WIDTH &&
										x < halfW + IMAGE_HALF_WIDTH + FRAME_WIDTH && y > halfH + IMAGE_HALF_WIDTH - FRAME_WIDTH && 
										y < halfH + IMAGE_HALF_WIDTH + FRAME_WIDTH)) {
					pixels[y * width + x] = 0xfffffff;
					//pixels[y * width + x] =0xDCDCDC;
				} else {
					// 此处可以修改二维码的颜色，可以分别制定二维码和背景的颜色；
					pixels[y * width + x] = matrix.get(x, y) ? 0xff000000 : 0xfffffff;
				}
			}
		}
 
		BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
		image.getRaster().setDataElements(0, 0, width, height, pixels);
 
		return image;
	}
 
	/**
	 * 把传入的原始图像按高度和宽度进行缩放，生成符合要求的图标
	 * 
	 * @param srcImageFile
	 *            --源文件地址
	 * @param height
	 *            --目标高度
	 * @param width
	 *            --目标宽度
	 * @param hasFiller
	 *            -- 比例不对时是否需要补白：true为补白; false为不补白;
	 * @throws IOException
	 */
	private static BufferedImage scale(BufferedImage srcImage, int height, int width, boolean hasFiller,int border, int padding) throws IOException {
		double ratio = 0.0; // 缩放比例
		//File file = new File(srcImageFile);
		//BufferedImage srcImage = ImageIO.read(file);
		srcImage = Yuanjiao.setRadius(srcImage,((srcImage.getWidth() + srcImage.getHeight()) / 6),border,padding);
		Image destImage = srcImage.getScaledInstance(width, height, BufferedImage.SCALE_SMOOTH);
		// 计算比例
		if ((srcImage.getHeight() > height) || (srcImage.getWidth() > width)) {
			if (srcImage.getHeight() > srcImage.getWidth()) {
				ratio = (new Integer(height)).doubleValue() / srcImage.getHeight();
			} else {
				ratio = (new Integer(width)).doubleValue() / srcImage.getWidth();
			}
			AffineTransformOp op = new AffineTransformOp(AffineTransform.getScaleInstance(ratio, ratio), null);
			destImage = op.filter(srcImage, null);
		}
		if (hasFiller) {// 补白
			BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
			Graphics2D graphic = image.createGraphics();
			graphic.setColor(Color.white);
			graphic.fillRect(0, 0, width, height);
			if (width == destImage.getWidth(null))
				graphic.drawImage(destImage, 0, (height - destImage.getHeight(null)) / 2, destImage.getWidth(null), destImage.getHeight(null), Color.white, null);
			else
				graphic.drawImage(destImage, (width - destImage.getWidth(null)) / 2, 0, destImage.getWidth(null), destImage.getHeight(null), Color.white, null);
			graphic.dispose();
			destImage = image;
		}
		return (BufferedImage) destImage;
	}
 
	
	public static BufferedImage createQRCode(String comment, BufferedImage backgroundImage,BufferedImage logoImage) throws WriterException, IOException{
		BufferedImage image = createQR(comment,450,450,backgroundImage);
		image = addWaterImage( logoImage, image, 213, 445);
		return image;
	}
	
	
	public static void main(String[] args) throws WriterException, IOException {
		File file = new File("d:\\20181110152455.png");
		BufferedImage backgroundImage = ImageIO.read(file);
		File file1=new  File("D:\\899261281076616340.png");
		BufferedImage logoImage=ImageIO.read(file1);
		BufferedImage code = createQRCode("http://www.bai.com",backgroundImage, logoImage);
		ImageIO.write(code, "png", new File("D:\\generator.png"));
		
	}

}
